// vararr2d.c -- 使用变长数组的函数
#include <stdio.h>

// #define TWO_DIMENSION 1

#ifdef TWO_DIMENSION
#define ROWS 3
#define COLS 4
int sum2d(int rows, int cols, int ar[rows][cols]);
int main(void)
{
    int i, j;
    int rs = 3;
    int cs = 10;

    int varr[rs][cs]; // 变长数组（VLA）
    for (i = 0; i < rs; i++)
        for (j = 0; j < cs; j++)
            varr[i][j] = i * j + j;

    printf("3x10 VLA\n");
    printf("Sum of all elements = %d\n", sum2d(rs, cs, varr));
    return 0;
}


// 带变长数组形参的函数
int sum2d(int rows, int cols, int ar[rows][cols])
{
    int r;
    int c;
    int tot = 0;
    for (r = 0; r < rows; r++)
        for (c = 0; c < cols; c++)
            tot += ar[r][c];
    return tot;
}
#else
// 二维数组有点复杂，还是先来个一维数组
int main(void)
{
    int i;
    int sz = 10;

    int varr[sz]; // 变长数组（VLA）
    for (i = 0; i < sz; i++)
        varr[i] = i + 1;

    printf("varr[1] = %d\n", varr[1]);
    return 0;
}

#endif